package com.aliyun.oss.utils;

import java.util.Base64;

public class Base64TranstrFile {
    public static byte[] extractContent(String base64Str){
        // 移除Base64前缀（如"data:image/png;base64,"）
        String base64Data = base64Str.split(",")[1];
        // 解码为字节数组
        return Base64.getDecoder().decode(base64Data);
    }

    public static String extractOriginalFilename(String base64Str){
        // 提取MIME类型（如"image/png"）
        String mimeType = base64Str.split(";")[0].split(":")[1];
        // 根据MIME类型映射文件扩展名
        String extension = getExtensionByMimeType(mimeType);
        // 生成原始文件名（这里简单处理为"avatar.扩展名"）
        return "avatar" + extension;
    }

    private static String getExtensionByMimeType(String mimeType){
        switch (mimeType){
            case "image/jpeg":
                return ".jpg";
            case "image/png":
                return ".png";
            case "image/gif":
                return ".gif";
            case "image/bmp":
                return ".bmp";
            case "image/webp":
                return ".webp";
            default:
                return ".jpg"; // 默认使用jpg扩展名
        }
    }
}
